#include <stdio.h>
#include <pthread.h>
#include <unistd.h>
#include <stdlib.h>

#define BUFFER_SIZE 5

int buffer[BUFFER_SIZE];
int count = 0;

//初始化锁
static pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;

//初始化条件变量
static pthread_cond_t cond = PTHREAD_COND_INITIALIZER;

void *producer(void *arg)
{
    int item = 1;

    while (1)
    {
        //加锁（获取互斥锁 拿不到等，拿到向下执行，其他线程不能进入）
        pthread_mutex_lock(&mutex);

        //如果缓冲区满了，等消费者读取
        while (count == BUFFER_SIZE)    
        {
            //暂停线程 等待唤醒
            pthread_cond_wait(&cond, &mutex);
        }
        
        //出while说明缓冲区没满，可以生产
        buffer[count++] = item++;

        printf("生产者向第%d位添加一个数据:%d\n",count - 1,buffer[count-1]);

        //唤醒等待的消费者,可以消费数据了
        pthread_cond_signal(&cond);

        //解锁（释放互斥锁，其他线程可以进入）
        pthread_mutex_unlock(&mutex);
    }
}

//消费者线程
void *consumer(void *arg)
{
    while (1)
    {
        //加锁（获取互斥锁 拿不到等，拿到向下执行，其他线程不能进入）
        pthread_mutex_lock(&mutex);

        //如果缓冲区为空，等生产者生产
        while (count == 0)
        {
            //暂停线程 等待唤醒
            pthread_cond_wait(&cond, &mutex);

        }
        //出while说明缓冲区有数据，可以消费
        --count;

        printf("消费者从第%d位读取出一个数据:%d\n",count,buffer[count]);

        //唤醒等待的生产者，可以生产数据了
        pthread_cond_signal(&cond);

        //解锁（释放互斥锁，其他线程可以进入）
        pthread_mutex_unlock(&mutex);
    }
}

int main()
{
    pthread_t producer_thread, consumer_thread;

    //创建生产者和消费者线程
    pthread_create(&producer_thread, NULL, producer, NULL);
    pthread_create(&consumer_thread, NULL, consumer, NULL);

    //等待两个线程结束
    pthread_join(producer_thread, NULL);
    pthread_join(consumer_thread, NULL);

    return 0;
}